#include "Dri_SPI.h"

/**
 * @brief 初始化SPI2
 */
void Dri_SPI_Init(void)
{
    GPIO_InitTypeDef GPIO_InitStructure;
    SPI_InitTypeDef SPI_InitStructure;
    
    /* 使能时钟 */
    RCC_AHB1PeriphClockCmd(RCC_AHB1Periph_GPIOC, ENABLE);
    RCC_APB1PeriphClockCmd(RCC_APB1Periph_SPI2, ENABLE);
    
    /* 配置SPI2引脚 */
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_InitStructure.GPIO_OType = GPIO_OType_PP;
    GPIO_InitStructure.GPIO_PuPd = GPIO_PuPd_UP;
    
    /* SCK */
    GPIO_InitStructure.GPIO_Pin = SPI2_SCK_PIN;
    GPIO_Init(SPI2_SCK_PORT, &GPIO_InitStructure);
    GPIO_PinAFConfig(SPI2_SCK_PORT, GPIO_PinSource13, GPIO_AF_SPI2);
    
    /* MISO */
    GPIO_InitStructure.GPIO_Pin = SPI2_MISO_PIN;
    GPIO_Init(SPI2_MISO_PORT, &GPIO_InitStructure);
    GPIO_PinAFConfig(SPI2_MISO_PORT, GPIO_PinSource14, GPIO_AF_SPI2);
    
    /* MOSI */
    GPIO_InitStructure.GPIO_Pin = SPI2_MOSI_PIN;
    GPIO_Init(SPI2_MOSI_PORT, &GPIO_InitStructure);
    GPIO_PinAFConfig(SPI2_MOSI_PORT, GPIO_PinSource15, GPIO_AF_SPI2);
    
    /* 配置片选引脚为推挽输出 */
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_OUT;
    GPIO_InitStructure.GPIO_OType = GPIO_OType_PP;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_InitStructure.GPIO_PuPd = GPIO_PuPd_UP;
    
    /* F_CS1 */
    GPIO_InitStructure.GPIO_Pin = F_CS1_PIN;
    GPIO_Init(F_CS1_PORT, &GPIO_InitStructure);
    GPIO_SetBits(F_CS1_PORT, F_CS1_PIN); // 默认拉高
    
    /* F_CS2 */
    GPIO_InitStructure.GPIO_Pin = F_CS2_PIN;
    GPIO_Init(F_CS2_PORT, &GPIO_InitStructure);
    GPIO_SetBits(F_CS2_PORT, F_CS2_PIN); // 默认拉高
    
    /* 配置SPI2 */
    SPI_InitStructure.SPI_Direction = SPI_Direction_2Lines_FullDuplex;
    SPI_InitStructure.SPI_Mode = SPI_Mode_Master;
    SPI_InitStructure.SPI_DataSize = SPI_DataSize_8b;
    SPI_InitStructure.SPI_CPOL = SPI_CPOL_Low;
    SPI_InitStructure.SPI_CPHA = SPI_CPHA_1Edge;
    SPI_InitStructure.SPI_NSS = SPI_NSS_Soft;
    SPI_InitStructure.SPI_BaudRatePrescaler = SPI_BaudRatePrescaler_8; // 约10.5MHz
    SPI_InitStructure.SPI_FirstBit = SPI_FirstBit_MSB;
    SPI_InitStructure.SPI_CRCPolynomial = 7;
    
    SPI_Init(SPI2, &SPI_InitStructure);

    SPI_Cmd(SPI2, ENABLE);
}

/**
 * @brief 启动SPI2
 */
void Dri_SPI_Start(void)
{
    SPI_Cmd(SPI2, ENABLE);
}

/**
 * @brief 停止SPI2
 */
void Dri_SPI_Stop(void)
{
    SPI_Cmd(SPI2, DISABLE);
}

/**
 * @brief SPI2发送并接收一个字节
 * @param data 要发送的数据
 * @return 接收到的数据
 */
uint8_t Dri_SPI_SwapByte(uint8_t data)
{
    /* 等待发送缓冲区空 */
    while (SPI_I2S_GetFlagStatus(SPI2, SPI_I2S_FLAG_TXE) == RESET);
    
    /* 发送数据 */
    SPI_I2S_SendData(SPI2, data);
    
    /* 等待接收缓冲区非空 */
    while (SPI_I2S_GetFlagStatus(SPI2, SPI_I2S_FLAG_RXNE) == RESET);
    
    /* 返回接收到的数据 */
    return SPI_I2S_ReceiveData(SPI2);
}

/**
 * @brief SPI2发送并接收数据
 * @param pTxData 发送数据指针
 * @param pRxData 接收数据指针
 * @param Size 数据长度
 */
void Dri_SPI_TransmitReceive(uint8_t* pTxData, uint8_t* pRxData, uint16_t Size)
{
    uint16_t i;
    for (i = 0; i < Size; i++)
    {
        pRxData[i] = Dri_SPI_SwapByte(pTxData[i]);
    }
}

/**
 * @brief 片选控制
 * @param cs_port 片选端口
 * @param cs_pin 片选引脚
 * @param state 1-拉高(取消选择), 0-拉低(选择)
 */
void Dri_SPI_CS_Control(GPIO_TypeDef* cs_port, uint16_t cs_pin, uint8_t state)
{
    if (state)
    {
        GPIO_SetBits(cs_port, cs_pin);
    }
    else
    {
        GPIO_ResetBits(cs_port, cs_pin);
    }
}
